#include<cstdio>
#include<cstring>
#include<string>
#include<sys/types.h>
#include<sys/wait.h>
#include<unistd.h>


using namespace std;

const int Bace=1024;
const int Argv=64;
const int Env=64;
//全局的命令行参数表
char *Gargv[Argv];
int Gargc=0;
//自己的环境变量
char *Genv[Env];

//当前shell的工作路径
char pwd[Bace];
//维护环境变量的pwd
char pwdenv[Bace];
//解释命令行的提示符
string GetUserName()
{
    string name = getenv("USER");
    return name.empty() ? "None" : name;
}

string GetHostName()
{
    string hostname = getenv("HOSTNAME");
    return hostname.empty() ? "None" : hostname;
}
string GetPwd()
{
    if(nullptr == getcwd(pwd, sizeof(pwd))) return "None";
    snprintf(pwdenv, sizeof(pwdenv),"PWD=%s", pwd);
    putenv(pwdenv); 
    return pwd;
}

string MakeCommandLine()
{

    char command_line[Bace];
    snprintf(command_line, Bace, "[%s@%s %s]# ",\
            GetUserName().c_str(), GetHostName().c_str(), GetPwd().c_str());
    return command_line;
}

void PrintCommandLine() // 1. 命令行提示符
{
    printf("%s", MakeCommandLine().c_str());
    fflush(stdout);
}


bool GetCommandLine(char command_buffer[], int size)   // 2. 获取用户命令
{
    // 我们认为：我们要将用户输入的命令行，当做一个完整的字符串
    // "ls -a -l -n"
    char *result = fgets(command_buffer, size, stdin);
    if(!result)
    {
        return false;
    }
    command_buffer[strlen(command_buffer)-1] = 0;
    if(strlen(command_buffer) == 0) return false;
    return true;
}

void ParseCommandLine(char command_buffer[], int len) // 3. 分析命令
{
    (void)len;
    memset(Gargv, 0, sizeof(Gargv));
    Gargc = 0;
    // "ls -a -l -n"
    const char *sep = " ";
    Gargv[Gargc++] = strtok(command_buffer, sep);
    // =是刻意写的
    while((bool)(Gargv[Gargc++] = strtok(nullptr, sep)));
    Gargc--;
}

// 在shell中
// 有些命令，必须由子进程来执行
// 有些命令，不能由子进程执行，要由shell自己执行 --- 内建命令 built command
bool ExecuteCommand()   // 4. 执行命令
{
    // 让子进程进行执行
    pid_t id = fork();
    if(id < 0) return false;
    if(id == 0)
    {
        //子进程
        // 1. 执行命令
        execvpe(Gargv[0], Gargv, Genv);
        // 2. 退出
        exit(1);
    }
    int status = 0;
    pid_t rid = waitpid(id, &status, 0);
    if(rid > 0)
    {
       // Do Nothing
       return true;
    }
    return false;
}

void AddEnv(const char *item)
{
    int index = 0;
    while(Genv[index])
    {
        index++;
    }

    Genv[index] = (char*)malloc(strlen(item)+1);
    strncpy(Genv[index], item, strlen(item)+1);
    Genv[++index] = nullptr;
}
// shell自己执行命令，本质是shell调用自己的函数
bool CheckAndExecBuiltCommand()
{
    if(strcmp(Gargv[0], "cd") == 0)
    {
        // 内建命令
        if(Gargc == 2)
        {
            chdir(Gargv[1]);
        }
        return true;
    }
    else if(strcmp(Gargv[0], "export") == 0)
    {
        // export也是内建命令
        if(Gargc == 2)
        {
            AddEnv(Gargv[1]);
        }
        return true;
    }
    else if(strcmp(Gargv[0], "env") == 0)
    {
        for(int i = 0; Genv[i]; i++)
        {
            printf("%s\n", Genv[i]);
        }
        return true;
    }
    return false;
}

// 作为一个shell，获取环境变量应该从系统的配置来
// 我们今天就直接从父shell中获取环境变量
void InitEnv()
{
    extern char **environ;
    int index = 0;
    while(environ[index])
    {
        Genv[index] = (char*)malloc(strlen(environ[index])+1);
        strncpy(Genv[index], environ[index], strlen(environ[index])+1);
        index++;
    }
    Genv[index] = nullptr;
}

int main()
{
    InitEnv();
    char command_buffer[Bace];
    while(true)
    {
        PrintCommandLine(); // 1. 命令行提示符
        if( !GetCommandLine(command_buffer, Bace) )   // 2. 获取用户命令
        {
            continue;
        }
        ParseCommandLine(command_buffer, strlen(command_buffer)); // 3. 分析命令

        if ( CheckAndExecBuiltCommand() )
        {
            continue;
        }

        ExecuteCommand();   // 4. 执行命令
    }
    return 0;
}
